package com.lollipop.programming.thread.juc;

import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

@Slf4j
public class CyclicBarrierDemo {
    public static void main(String[] args) {
        // 创建CyclicBarrier，参与线程数为3，所有线程到达屏障后执行回调
        CyclicBarrier barrier = new CyclicBarrier(3, () -> {
            log.info("所有线程已到达屏障点，开始执行下一阶段");
        });

        ExecutorService executor = Executors.newFixedThreadPool(3);

        // 创建3个任务
        for (int i = 1; i <= 3; i++) {
            final int taskId = i;
            executor.submit(() -> {
                try {
                    log.info("任务{}第一阶段开始执行", taskId);
                    Thread.sleep(1000 * taskId);
                    log.info("任务{}第一阶段完成，等待其他线程", taskId);

                    // 等待其他线程
                    barrier.await();

                    log.info("任务{}第二阶段开始执行", taskId);
                    Thread.sleep(500 * taskId);
                    log.info("任务{}第二阶段完成", taskId);

                    // 可以再次使用同一个barrier
                    barrier.await();
                    log.info("任务{}所有阶段完成", taskId);
                } catch (InterruptedException | BrokenBarrierException e) {
                    log.error("任务{}执行失败:{}", taskId, e.getMessage());
                }
            });
        }

        executor.shutdown();
    }
}